package com.ccnode.codegenerator.taikaoback.util;

import lombok.Data;

import java.io.Serializable;
import java.time.LocalDateTime;
import java.time.ZoneOffset;
import java.util.HashMap;
import java.util.Map;

/**
 * 通用响应类
 * @param <T> 响应数据类型
 */
@Data
public class R<T> implements Serializable {
    private static final long serialVersionUID = 1L;

    /**
     * 状态码
     */
    private Integer code;

    /**
     * 提示信息
     */
    private String message;

    /**
     * 数据
     */
    private T data;

    /**
     * 时间戳
     */
    private Long timestamp;

    /**
     * 构造函数
     */
    public R() {
        this.timestamp = LocalDateTime.now().toInstant(ZoneOffset.of("+8")).toEpochMilli();
    }

    /**
     * 成功返回结果
     *
     * @param data 返回数据
     * @param <T> 数据类型
     * @return 通用返回结果
     */
    public static <T> R<T> success(T data) {
        R<T> r = new R<>();
        r.setCode(200);
        r.setMessage("success");
        r.setData(data);
        return r;
    }

    /**
     * 成功返回结果
     *
     * @param message 提示信息
     * @param data 返回数据
     * @param <T> 数据类型
     * @return 通用返回结果
     */
    public static <T> R<T> success(String message, T data) {
        R<T> r = new R<>();
        r.setCode(200);
        r.setMessage(message);
        r.setData(data);
        return r;
    }

    /**
     * 失败返回结果
     *
     * @param code 状态码
     * @param message 提示信息
     * @param <T> 数据类型
     * @return 通用返回结果
     */
    public static <T> R<T> error(Integer code, String message) {
        R<T> r = new R<>();
        r.setCode(code);
        r.setMessage(message);
        Map<String, String> data= new HashMap<>();
        data.put("message", message);
        r.setData((T) data);
        return r;
    }

    /**
     * 失败返回结果
     *
     * @param message 提示信息
     * @param <T> 数据类型
     * @return 通用返回结果
     */
    public static <T> R<T> error(String message) {
        return error(400, message);
    }

    /**
     * 失败返回结果（带数据）
     *
     * @param message 提示信息
     * @param data 返回数据
     * @param <T> 数据类型
     * @return 通用返回结果
     */
    public static <T> R<T> error(String message, T data) {
        R<T> r = new R<>();
        r.setCode(400);
        r.setMessage(message);
        r.setData(data);
        return r;
    }

    /**
     * 失败返回结果（带状态码和数据）
     *
     * @param code 状态码
     * @param message 提示信息
     * @param data 返回数据
     * @param <T> 数据类型
     * @return 通用返回结果
     */
    public static <T> R<T> error(Integer code, String message, T data) {
        R<T> r = new R<>();
        r.setCode(code);
        r.setMessage(message);
        r.setData(data);
        return r;
    }

    /**
     * 未授权返回结果
     *
     * @param <T> 数据类型
     * @return 通用返回结果
     */
    public static <T> R<T> unauthorized() {
        return error(401, "未授权");
    }

    /**
     * 权限不足返回结果
     *
     * @param <T> 数据类型
     * @return 通用返回结果
     */
    public static <T> R<T> forbidden() {
        return error(403, "权限不足");
    }

    /**
     * 资源不存在返回结果
     *
     * @param <T> 数据类型
     * @return 通用返回结果
     */
    public static <T> R<T> notFound() {
        return error(404, "资源不存在");
    }

    /**
     * 服务器内部错误返回结果
     *
     * @param <T> 数据类型
     * @return 通用返回结果
     */
    public static <T> R<T> serverError() {
        return error(500, "服务器内部错误");
    }
} 